FROM registry.openanolis.cn/openanolis/cuda:12.1.1-23-runtime

LABEL \
    maintainer="OpenAnolis AI SIG" \
    org.opencontainers.image.title="jupyterlab" \
    org.opencontainers.image.version="4.0.5-23" \
    org.opencontainers.image.date="20231206"

RUN dnf install libcudnn -y
RUN dnf install pytorch torch-vision -y
RUN dnf install tensorflow2 -y
RUN dnf install jupyterlab nodejs -y

CMD ["/bin/bash"]
